Goto

Collaborating Authors

 efficient kv cache quantization


ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification

Neural Information Processing Systems

KV cache stores key and value states from previous tokens to avoid re-computation, yet it demands substantial storage space, especially for long sequences. Adaptive KV cache compression seeks to discern the saliency of tokens, preserving vital information while aggressively compressing those of less importance. Additionally, the compression process introduces excessive overhead, substantially increasing memory burdens and the generation latency. First, we construct a strong baseline for quantizing KV cache. Through the proposed channel-separable tokenwise quantization scheme, the memory overhead of quantization parameters are substantially reduced compared to fine-grained groupwise quantization.